# data from train dir               <-- Not good 
data = load_data()