Updating lesson 5, fixing type in 4