add if statement, and note of chapter 4.