Update ProbController.py