Update bagging.py