небольшие правки в коде