Small refinements to make exxercise 2 & 3 neater and add comments