Downstream code separated for data and MC