fixed issue 11
Merge pull request #21 from ninayang/master

fixed issue 11. Good!