Add highlighting for image buttons