Minor script update