Add after render callback for edit views