Добавил тесты

Покрыл тестами 80% всего кода :)