add prompt_env function