Final working version, except the highlight one not working for all
words.