Step 2 - updated

Updated to show code for generating figures