Update for week 4 work

Update for week 4 work