Update main.py

fix graph function, make a legend, add axis lines, add zoom to graph() function