Modified Ensemble class