diff --git a/count_sum_pandas.py b/count_sum_pandas.py
index 09da9a6..9fca5a3 100644
--- a/count_sum_pandas.py
+++ b/count_sum_pandas.py
@@ -1,4 +1,4 @@
-import pandas as pd
-
-def count_sum(df: pd.DataFrame) -> pd.DataFrame:
-    return df.groupby('Товар')['Количество'].sum().reset_index()
\ No newline at end of file
+import pandas as pd
+
+def count_sum(df: pd.DataFrame) -> pd.DataFrame:
+    return df.groupby('Товар')['Количество'].sum().reset_index().set_index('Товар')
\ No newline at end of file