Looks great! If you don't mind, could you tell me the direction you're going with this if you're going to release the updated source to the public? I'm working on editing things as well to create more gameplay and make it more fun in general, so I don't want there to be a duplication of effort with both of us working on the same features.
I'll release my edits as well, if you'd like.