Combine code for both models in single annotated script