Выпрямил немного код

Правил пару файлов