Fixing rubocop failures