Add tests for hw3