Add test suite code