Create einstein_problem.py