added test class
More Unit Testing