Added a figure and made some grammar fixes