Fix bugs.

Поскольку по пока не ясной мне причине не получается пушить прямо в Тошин репозиторий, открываю пулреквест с исправлением багов и улучшениями.