add test code
chapter 4 learn part 1