Change argument to be docId