Added graphs and main code