Tinkering with models - made it work