Add types for particle momentum