Week six assignment 1