Make Python version check also work for Python 4+