Binary to Grey Code and Grey to Binary in Verilog using mode switch

Question

I am implementing Code converter with a mode switch such that mode 0 implies binary to grey code and mode 1 implies grey to binary conversion. My design and testbench is as