Added docs
Fixed bug
Merge pull request #83 from johnttan/fixBug

Fix bug