modified import model code