Rewrote higlight method to be much less glitchy