Ученик попросил ИИ решить задачу на кинетическую энергию. ИИ сгенерировал код, но в нём есть логическая ошибка. Исправьте решение так, чтобы оно соответствовало