Rewrite the test coverage exercise in lab 2