Add setup.py customizations and fixes