Cleaning up code with rubocop