Minor comments improvement.