adding in lecture 5