Corrected the plotting