Added argument to highlight function.

Before commit, page build error. After commit? We'll see.