[PATCH 2/3] Update CPS calculation with new data structures