Add set state script