Add controls for edit mode to table elements