Added class definition to example