updated models to match import file