Added basic test suite code