Рефакторинг

Исправление мелких багов