minor script polish