Next step: make it classify the next word.