Add model to __all__