Немного изменили код