Heat-coloring the branches + other small improvements in tree visualization