Add a flag to limit how many contexts in total we explore.