Tweaked code for rubocop
Merge pull request #14 from jinlee0908/lee

Tweaked code for rubocop