small corrections in visualisation