Complete quiz 4 code