Added basic test class to enable code execution