Install script improvements