Install script improvement