чуть изменен код + добавлен тест