Оптимизация кода

Сделал компактней