Add edit function to story photos and videos