Visualisation almost finished