Add the option to prefill the context