Fixed bug not reacting when predict button is pressed