Code for lecture 3