Add p2m in compute_potential.pyx