Add rubocop
Clean code to pass rubocop.