Improved version of install.py