Create axis.py

File to draw the axis