Fixing issue 11.
Merge pull request #12 from codywd/master

Fixing issue 11.