added testing for the two models