Захардкодил прохождение тестов.
Избавился от дублирования кода.