Ensure model classes accept a max_mem input

Created by: CompPhysChris

The BE_SHO_model does not. Other models should be checked as well.