Added graphs and code updates