Update visualization tutorial

[coruscating]

As reported on Slack, the customize plotting section of the visualization tutorial is out of date and doesn't run with the current codebase. Ideally this code should be executed along with the rest of the tutorial so we can prevent this kind of regression.