WIP: add MultiStageReg2

Change-Id: If17c93f2f9c48ced175c98282c0b99ccec13bd71