fixed a little bug and add an improvement
Merge pull request #6 from pangeran-bottor/master

fixed a little bug and add an improvement