Refactor visualisations to not require a section