Wrote up code for PMI model. not working yet :(