Update pso_examples.py