Fixed missing imports in model.py.
Fixed main routine to run all types of models.