Make visualization configurable