Add boosting methods to the plotting function