Fixed #4 as specified on bug description
Merge pull request #5 from Tlantic/v0.4.1

Fixed #4 as specified on bug description