prediction class added