update ensembles to take in context