Modify again in lesson 6