Edit note functionality works