Fixing rubocop errors