Part 5: Setting up Word Prediction