Thanks authors for your great work in this repo. I was looking at your wrapped normal pdf function, and assuming x and p_ are 3-dimensional, it seems as though the wrapping is only done over translations of the form [T,T,T] instead of [Tx,Ty,Tz] for integers T, Tx, Ty, Tz. I would think [Tx,Ty,Tz] would be more accurate - seems like a minor detail but I was hoping you could comment on it. Thanks!