Fix code.
Merge pull request #4 from chasechocolate/master

Fix code.