Minor indent and refactor
Merge pull request #4 from samebchase/refactor

Minor indent and refactor