Fixing multiclass input set from 3 to 5.