Add tests for week 3