Вносим небольшие корректировки в код