PID upgrade in the endcap