Added a bit more code for the lecture to come.