Rework getClasses again