updated generator for 2 particles test