Adding lesson 2 in class files