Improve install.py