Fixing rubocop rules