Code cleanup for viz.