From fe4390dca26e86f47a09f4d8678ba89395892b18 Mon Sep 17 00:00:00 2001
From: jwen2 <31702182+jwen2@users.noreply.github.com>
Date: Wed, 1 Nov 2017 23:39:08 -0400
Subject: [PATCH] After many hours, I now officially hate Pickles.

---
 counter.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/counter.py b/counter.py
index 38ec816..e28ab97 100644
--- a/counter.py
+++ b/counter.py
@@ -30,7 +30,21 @@ def update_counter(file_name, reset=False):
     >>> update_counter('blah2.txt')
     2
     """
-    pass
+    if exists(file_name) and reset == False:
+        Storage = open(file_name, 'rb+')
+    else:
+        Storage = open(file_name, 'wb')
+    if reset == True:
+        Storage = open(file_name, 'wb')
+        dump(1, Storage)
+        Storage.close()
+        Input = open(file_name, 'rb+')
+        return load(Input)
+    else:
+        Counter = int(load(Storage)) + 1
+        Storage.seek(0,0)
+        dump(Counter, Storage)
+        return Counter
 
 if __name__ == '__main__':
     if len(sys.argv) < 2: