Tidy up install script