added highlighting functionality