Massive changes to graph.py and Input folder