Minor changes, including adding some figures to show the simulation