Added list to store models
Now you can make predictions on the test set