Adding edit capabilities