Update axis.py