added Kdiag to parallel implementation