Chapter 2 Code chunk 2 for HW1