Improving the Knowledge Gradient Algorithm