Add in class learning