small corrections in visualisation again