Fixed code example.

The code example should define the class as it is mentioned before.