Оптимизация

Оптимизировать и упростить код. Вынести дублирующийся код в функции