Simplified version-finding in setup.py so it also works nicely with Python 3.