Adding dataset list as argument for create_model and create_ensemble