Update PMF.py