Added test cases for some classes