Added initial test class