Adding more functionality to the Python model.