Install script improved