Adding methods to remove buttons and edit_line elements