1 files changed, 9 insertions, 9 deletions
diff --git a/src/main/TweetCounter.java b/src/main/TweetCounter.java
index 6c1397a..c630cda 100644
--- a/src/main/TweetCounter.java
+++ b/src/main/TweetCounter.java
@@ -1,12 +1,12 @@
 package main;
 
+import com.google.gson.JsonObject;
+import com.google.gson.JsonParseException;
 import java.util.Date;
 import java.util.HashSet;
 import java.util.Set;
 import java.util.logging.Level;
 import java.util.logging.Logger;
-import org.json.JSONException;
-import org.json.JSONObject;
 import provider.ResultListener;
 
 /**
@@ -29,14 +29,14 @@ public class TweetCounter implements ResultListener {
     }
 
     @Override
-    public void tweetGenerated(JSONObject obj) {
-        tweetCount++;
+    public void tweetGenerated(JsonObject obj) {
         try {
-            JSONObject userObj = obj.getJSONObject("user");
-            String screen_name = userObj.getString("screen_name");
-            users.add(screen_name);
-        } catch (JSONException ex) {
-            LOGGER.log(Level.WARNING, "Profile is missing data", ex);
+            JsonObject userObj = obj.getAsJsonObject("user");
+            String screen_name = userObj.get("screen_name").getAsString();
+            tweetCount++;
+            users.add(screen_name); 
+        } catch (JsonParseException ex) {
+            LOGGER.log(Level.WARNING, "Profile is missing data", ex);            
         }
     }