[Bugfix] Colouring for visualisation