added policy learning code