Added edge case for returning error on unsuitable predictions
Added code to deal with edge case