Помогите пожалуйста сделать программу "Генератор задач по физике" Нужно на питоне если что

Ты можешь сделать что-то подобное с другой формулой. Я не силен в физике, поэтому напиши программу на формулу скорости.

import random

def generate_problem():
u = random.randint(0, 50)
a = random.randint(1, 10)
t = random.randint(1, 10)

v = u + a * t

problem = f"Найти итоговую скорость тела, если начальная скорость {u} м/с, ускорение {a} м/с^2 и время движения {t} с."
answer = f"Итоговая скорость тела равна {v} м/с."

return problem, answer

num_problems = 5

for i in range(num_problems):
problem, answer = generate_problem()
print(f"Задача {i+1}:")
print(problem)
print(answer)