Velocity fields in 3d

I was checking this great tutorial

which works in 2d, what would need to be done to make it work in 3d, so that the particles also move in depth ?
thank you :slight_smile: