Adding editing functionality.