working with function in Python Chapter 3