ENH: Update lesson 2 and lesson 4