fig, ax = plt.subplots()
ax.axis('square')