Внес изменения

Добавил тесты, переделал в lib