Remove unnecessary code to use other element. Not just DIV