Small fixes

Поправки в коде