Updated script to allow for Python version selection