Added some testing of models