Figures for the simulation part updated