Minor visualization code cleanup.