fixed issue 2
Merge pull request #15 from adisa/master

fixed issue 2