Fixed one initialization bug - but our reward / loss function is enormous. Probably need to normalize it to be [0,1]