перехерачил классы, стало логичнее, но тесты не проходят