Fixing Rubocop