Fixed up visualisation