Fixing Rubocop complaints