Chapter 4 Code chunk 2 for HW1