Ben Hohnke

This doesn't actually work - there was a flaw in my testing. Working on a proper fix

> thanks for your contribution, could you add some tests to avoid regressions? Test cases added Also updated test_models.py to allow multiple test cases for each test. It might not...

Sounds good!