Add test bench