fixed a couple of small bugs with the visualisation