Added some tests with models including a bug :(