Make final split-vi plot pretty

Minor update to the example script