Доработки и тесты
Добавил тестирование