Editing setup.py for functionality