Updated for lesson 10