2 теста проходит + рефакторинг кода