Updates setup.py to include sub directories