Updates to enable the PU forest running