Added tree button handlers