Code up to lesson 6