Added token prediction to the new question dialog