Completed plotting code and figures