Fixed the issue with visualization changing the random seed, working on policy iteration