Working test class

got it working