Class prediction project