Highlight functionality finished:)