Add new classes to setup.py