Exercise 2 for class project