update to match new code in lesson 3