Clean up with rubocop