Исправление отображения ошибок + рефакторинг