Fix example code for exrecise 3