Add shift left and shift right to main stage proof