Testing PPO's through displacement formalism

Still not working properly
Corrected PPO error

Changed 1 to ans[[i,j]] in if statement, fixed everything