Small update to classes.