fixing new dataset formatting

main.py

@@ -41,6 +41,8 @@ def main():
     # New dataset
     else:
         df = new_df
+        df['new'] = True
+        df['updated'] = False
         new_rows = len(new_df)
     dataset['train'] = Dataset.from_pandas(df, preserve_index=False)