Adding edit capability