Better explanation of policy iteration
Minor improvements