tweaks to tests and visualisation