Update lesson1 code