Make python support all model types