Edit lesson 4 a bit more