Added ability to run only local context component of prob.