Improve the installation script