Add missing modules to setup.py