comment layout optimise of graph
pep8 fix