New visualisations and fixes