Добавил тесты, исправил ошибки