Implement pseudo element :after & remove unnecessary funtions