Update RefactoredModels.py

added assertions for the intermediary models to make sure they take the right types