Improve the python driver installer script