Plotting the variable rates model for Continuous trait variation [BayesTrait]

[Jigyasa3]

Thank you so much for making all the lectures available online! I have learnt alot from them!

I was wondering if its possible (R package already available) to plot the variation in a tree using variable rate model?
I ran BayesTrait and found that there is a strong evidence of variation of trait with time. The analysis also generates a Output tree file. But it only stretches/shrinks the tree with different rate of change. I wanted to ask if there is a way to map it on a tree using heatmap? [like contMap in phytools]