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