How to do model imports for DR2