Add figure + better comments.