Tweaked for rubocop
Merge pull request #12 from jinlee0908/lee

Tweaked for rubocop