small cleanup of viz, plot now refresh when rerun.