Added option to graph simulation