Update decision-tree button class