add code for slide 4