Modified plot so matches work