diff --git a/competition/classifier.py b/competition/classifier.py
index 11e4838..ef5aa56 100644
--- a/competition/classifier.py
+++ b/competition/classifier.py
@@ -42,4 +42,4 @@ data['destinationIPAddress'] = data['destinationIPAddress'].apply(bin_to_ip)
 print(data)
 
 # 3. Save output file
-pickle.dump(pd, open('output.csv', 'wb'))
+data.to_csv('output.csv', index=False)