Update demo_floating_axes.py