small changes
Merge pull request #5 from forrest-mao/suggestion

small changes