Update pso.cc