Added first pass at policy gradient