Polished the toki highlighter

Signed-off-by: Frank Mertens <frank@cyblogic.de>