Update xgboost.py