Add PI tracking