add function to test calss