Update setup.py for egg creation