Edits to PSO trainer.