Fixes #228; disease models now have source object rather than string