Update week 4 exercise