Лёгкий причёс кода.
Разделил классы по файлам.