update class definition to match examples