fixed small bugs
Merge pull request #5 from thebrianzeng/master

fixed small bugs