Немного подправил тесты, покрыл тестами немного
Исправил небольшие ошибки.