Fixed rubocop issues