Add #replace_element helper.
Update TODO.