added proper arguments for stretch_ensemble