Added code to reapply DOM changes

Does not work properly yet, there's still issues finding the specific
elements to add on to.