ENH: Add class_weights to trees. WIP.