set up for lessson 4
save final code for lesson 3