Extending create_predictions to ensemble predictions and more tests