Visualisation Bug Fix and Tweaks