add tests to exercise4