Fixing rubocop complaints