Made some improvements to the installation script