Added some methods to test models