Re-add prediction update button