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

Исправили ошибки и улучшили работу.