tuned the figures, especially figure sizes for publication, and added more explanations to the notebook.