fixes #3

fixes #3 no tests
Awesome job! Go ahead and merge this back into master.