add code to display word predictions