added button action to predict button