Add class iterator support