return full model
retrun correct model on pu