diff --git a/correlation.py b/correlation.py
index 4538354..fbaa828 100644
--- a/correlation.py
+++ b/correlation.py
@@ -93,11 +93,14 @@ if __name__ == "__main__":
         print(f'Average SWAP score: {np.mean(swap_score)}')
         print(f'Elapsed time: {end_time - start_time:.2f} seconds')
 
-        results.append([np.mean(swap_score), acc])
+        results.append([np.mean(swap_score), acc, i])
+
+    results = pd.DataFrame(results, columns=['swap_score', 'valid_acc', 'index'])
 
-    results = pd.DataFrame(results, columns=['swap_score', 'valid_acc'])
     print()    
     print(f'Spearman\'s Correlation Coefficient: {stats.spearmanr(results.swap_score, results.valid_acc)[0]}')
+    results.to_csv('swap_results.csv', float_format='%.4f', index=False)
+