Making some tweaks

Minor tweaks and labeled last figure.