refactor
Merge pull request #4 from acprimer/fixbug-0.1

refactor