add lesson 3 code until Unit setting