Improved installation script