Updates to sample script