added class and test class