Fixed #2.
Merge pull request #5 from brendanbates89/master

Fixed #2.