Updated with lesson 4 content