added other models to the testing script