Correcting policy matrices
Column vectors for uniformity
Using policy matrices
Code formatting