Improve lesson 3 4