Update visualisatie code