you should add highlighting function

good idea id like that