Fix python version check
Tweak setup.py