updated models to fix image error