update kSZ code