маленькие изменения в коде :3