KD: Added test cases for models