Create weight_problem.py