Adding in what is necessary for MMS problems/sources