Working on PSO