Added basic test code