Little class change