write model base class , now the model choice can work