Рефакторинг кода. Убрал лишнее, упростил.