Porting to Chisel 2.0