Fixed rubocop warnings