Added new figures and ran through all comments