small changes to PEPS_plots