Changed figure and added next steps.