હાર્દિક પંડ્યાની વાપસીથી ટીમનું સંતુલન વધુ સારું થયું છે: કોહલી