fixes in lesson 9