Plots and codes fixed