Purity as function of momentum