Add tests from week 4