Addition to graph visualisation
Bug fix