Added a comparison statement to set the predictionLabel text