Setup rubocop
Tweak code