Added a figure along with figure code and other small edits.