Fixed bug.
Merge pull request #11 from stroud109/master

Fixed bug.