import roboticstoolbox as rtb

puma = rtb.models.DH.Puma560()
env = puma.plot(puma.qr, color="gray", block=False)
puma.plot(puma.qn, env=env, color="red", name=False, block=False)