fixed visualisation colouring