Fix the class names in the example code