fixes #3 added :)

Awesome job! Go ahead and merge this back into master.