Added code for square highlight
Added square highlight