Enhance install.py script