update script a little