small change to PEPS plots