added a lot for lesson 4