pygame working and policy iteration debugging