Fix offences
Merge pull request #22 from leafac/clean-up

Fix rubocop offences