Add test codes