diff --git a/model_search_upload.py b/model_search_upload.py
index 51467ac..6316b18 100644
--- a/model_search_upload.py
+++ b/model_search_upload.py
@@ -82,6 +82,7 @@ def uploat_to_mlflow(temp_dir, **context):
         print("Uploading model search results")
         df = pd.read_csv(os.path.join(temp_dir, 'pd.csv'), index_col=0)
         dct = df.to_dict()
+        print("Got following data", dct)
         metrics=['mean_test_score', 'mean_fit_time']
 
         for i, p in enumerate(dct['params'].values()):