fixed output graph, modified testcase and README