minor fixes
lesson 4 done + minor fixes