[ENH] UFEM: Brownian motion for particles and fix

Attempted fix for NaN when 0 particle concentration