adds functionality to edit controller