Added basic test class