Changes to predict and if checks returns