matplotlib in python - drawing shapes and animating them,plt.show()