Small refactor to install.py