Add edit files feature