Add set_env