ДЗ(2)

Митрофанов Кирилл

Хотя нет, с вариантом всё правильно.
Тогда просто добавьте тестов.