[AI] Lab 2 PSO redo (still doesn't work well)