Updated setup.py for compatibility