Installation script improvements