Visualizing a tree with nodes of different colors with anytree

Question

I am looking for a simple way to construct and draw a tree (on google colab).

Importantly, I would like to have nodes of different colors and shapes. Ideally, I would like something as follows.

from anytree import Node, RenderTree
from anytree.exporter import DotExporter
from IPython.display import Image


# construct tree
ceo = Node("CEO") #root

vp_1 = Node("VP_1", parent=ceo, color="red")
vp_2 = Node("VP_2", parent=ceo)

gm_1 = Node("GM_1", parent=vp_1, shape="square", color="red")
gm_2 = Node("GM_2", parent=vp_2, shape="square")

m_1 = Node("M_1", parent=gm_2)

# draw tree
DotExporter(ceo).to_picture("ceo.png")

# show image
Image('ceo.png')

As color and shape are not real arguments of Node, this code currently generates the following image. I would like VP_1 and GM_1 to be red, and GM_1 and GM_2 to be squares.

Thank you very much for your help!