Переписал тесты, рефакторинг