Fixed issue #2
Merge pull request #3 from jupiter24/issue2

Fixed issue #2