Tweak installation script