Исправление косяков

Косяки + визуальные изменения