Adding some more test cases for the model extractor