Added test cases for site and class models