HDLBits-Adder subtractor

a circuit that can do two operations: (a + b + 0) and (a + ~b + 1). 

module top_module(
    input [31:0] a,
    input [31:0] b,
    input sub,
    output [31:0] sum
);
    
    wire cout;
    wire [31:0] w1;
    
    assign w1 = b ^ {32{sub}};
    
    add16 add16_0(a[15:0],w1[15:0],sub,sum[15:0],cout);
    add16 add16_1(a[31:16],w1[31:16],cout,sum[31:16],);

endmodule