Сначала сохрани логику
При обработке кода приоритет — сначала сохранить логику, затем при необходимости предлагать улучшения.