Added working graph for end of simulation