mini refactoring in visualisation