Get rid of extra class in example