Finish with exercise 3 (we modified in lib/Week5FromLecture)