Updated register and flag visualisations