HW for week 3 of Cal Berkeley Saas course