base model improvment