Added UI to assign classes