Correct Error in predictNextWord

Correct error in predictNextWord selecting incorrect columns.