Cleanups and added interface to FermiDiffuseModel.