add tests to lesson 4