Add the ability to switch trees