Fixed code on slide 16 and 18 issue #1