Graph visualisation fixed