made test pass for lesson 1 + 2