Add methods to get all classes, and add and remove classes