Added some minor changes
Merge pull request #5 from jameshwang/james/make_pretty

Added some minor changes