Add focus state for before and after elements and close button