Fixing rubocop warnings