Add some code for after pseudo element