added some testing for models