diff --git a/correlation/NAS_Bench_201.py b/correlation/NAS_Bench_201.py
index 9bab538..733a4bb 100644
--- a/correlation/NAS_Bench_201.py
+++ b/correlation/NAS_Bench_201.py
@@ -1,14 +1,11 @@
 import argparse
-import os
-
-import time
+import pickle
+import random
 
 from foresight.dataset import *
 from foresight.models import nasbench2
-from foresight.pruners import predictive
 from foresight.weight_initializers import init_net
 from models import get_cell_based_tiny_net
-import pickle
 
 
 def get_score(net, x, device, measure='meco'):
diff --git a/correlation/compute_rho.py b/correlation/compute_rho.py
index 6558891..792ce36 100644
--- a/correlation/compute_rho.py
+++ b/correlation/compute_rho.py
@@ -1,15 +1,15 @@
 import pandas as pd
 import pickle
 
-# path = 'result/sss_cf10_meco.p'
-path = 'nb2_sss_cf10_seed42_dlappoint_dlinfo1_initwnone_initbnone_1.p'
+path = 'result/sss_cf10_meco_opt.p'
+# path = 'nb2_sss_cf10_seed42_dlappoint_dlinfo1_initwnone_initbnone_1.p'
 meco = []
 accs = []
 with open(path, 'rb') as f:
     while True:
         try:
             fl = pickle.load(f)
-            meco.append(fl['meco'])
+            meco.append(fl['meco'][0])
             accs.append(fl['testacc'])
         except:
             break