new changes

src/main/java/com/zipcodewilmington/phonebook/PhoneBook.java

@@ -1,11 +1,6 @@
 package com.zipcodewilmington.phonebook;
 
-import java.sql.SQLOutput;
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.Map;
 import java.util.TreeMap;
-import java.util.Collection;
 
 /**
  * Created by leon on 1/23/18.
@@ -13,41 +8,40 @@ import java.util.Collection;
 
 public class PhoneBook {
 
-    TreeMap <String, String> phoneBook = new TreeMap <String, String>();
+    TreeMap<String, String> phoneBook = new TreeMap<String, String>();
 
     public PhoneBook() {
 
-
     }
 
     public boolean hasEntry(String name) {
 
+        if (phoneBook.containsKey(name)) {
 
-
-        return phoneBook.containsKey(name);
+            return true;
+        }
+        return false;
     }
 
-
     public void add(String name, String phoneNumber) {
 
+        if (this.hasEntry(name)) {
 
-        phoneBook.put(name, phoneNumber);
-
+            this.phoneBook.get(name);
+        }
 
+        phoneBook.put(name, phoneNumber);
 
     }
 
     public String lookUp(String name) {
 
-
         return phoneBook.get(name);
 
-
     }
 
     public void remove(String name) {
 
-
         phoneBook.remove(name);
 
     }
@@ -56,9 +50,9 @@ public class PhoneBook {
 
         String result = "";
 
-        for(String keyName : phoneBook.keySet()) {
+        for (String keyName : phoneBook.keySet()) {
 
-            if(phoneBook.get(keyName).equals(phoneNumber)) {
+            if (phoneBook.get(keyName).equals(phoneNumber)) {
 
                 result = keyName;
 
@@ -70,17 +64,11 @@ public class PhoneBook {
 
     public void listNamesAndNumbers() {
 
-        for(String keyName : phoneBook.keySet()) {
-            System.out.println(keyName + " " + phoneBook.get(keyName));
-        }
-
+        for (String keyName : phoneBook.keySet()) {
 
+            System.out.println(keyName + " " + phoneBook.get(keyName));
         }
-
-
-
-
-
+    }
 }
 
 

src/test/java/com/zipcodewilmington/phonebook/PhoneBookTest.java

@@ -22,7 +22,6 @@ public class PhoneBookTest {
         String name = "xzavia";
         String phoneNumber = "302-399-3694";
         phone.add(name, phoneNumber);
-
         boolean actual = phone.hasEntry(name);
 
         Assert.assertTrue(actual);