Imporve tree plot