Re-working p_loop and corresponding functions to not depend on the data ordering

[manncz]

The estimator should not depend on the ordering of the data, so we should consider re-working the code so that we reshape the data to have the treated unit first at the beginning of the code and never use the original arrangement.