Update the design doc

Update for the design