Fixed issue#2.
Merge pull request #4 from StevePo/issue_2

Fixed issue#2.