Updated the tests for lesson 2.