Update stats_problem.py