Updated setup.py.
Added support for subclassed models.