update code.
Updated lesson 4 code.