Add block highlighting