Add test2 code