adding class to example code