Modify to cover pseudo element :before and :after with display = block