Update set environment script