Code review

Looks good, a couple of minor comment. Very please to see you already improved the descriptions for the various figures.