Add TreeSelection signal and other fixes