Minor updates
Updates title
Merge pull request #20 from joshjeong/refactor

Refactor