Improvements to the install script