Changed to div tag with class