Fixed Python model. Now it works.