Create stats_problem.py