Небольшие правки.
Небольшие изменения и пара лишних тестов.