оптимизация

Неьольшая оптимизация и исправление мелкого бага