added test classes for new schema