Added edit capabilities