Add PMI Model class