update to end of lecture 3 demo