Minor updates to figures and notebook.