Install script tweaks