Update to lesson 3