plt.savefig('graph.png')
plt.show()