Lecture 4 exercise