PSO working in 2D