Improve the implementation slightly when coding against the new online judge.