Updated setup.py so it works, but still need to work on py2exe