first draft completed but still need extra test cases

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

@@ -12,6 +12,7 @@ public class PhoneBook {
     //instance variable
     private TreeMap<String, String> treeMap;
 
+
     // void "no arguments" constructor -creates empty map
     public PhoneBook() {
         this.treeMap= new TreeMap<String, String>();
@@ -24,9 +25,11 @@ public class PhoneBook {
 
     //add a name & number entry
     public void add(String name, String number) {
+
         if(!treeMap.containsKey(name)) {
             treeMap.put(name, number);
         }
+
     }
 
     //remove a name & number entry
@@ -40,13 +43,8 @@ public class PhoneBook {
 
     //find phone number lookup by name
     public String lookup(String name) {
-        String look = "";
-        if (treeMap.containsKey(name)) {
-            look = treeMap.get(name);
-        } else {
-            look = name + " is not in database";
-        }
-        return look;
+        String number = treeMap.get(name);
+        return number;
     }
 
     //print out all of the entries in PhoneBook

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

@@ -13,22 +13,29 @@ public class PhoneBookTest {
     public void testDefaultConstructor() {
 
         PhoneBook book = new PhoneBook();
-        book.add("Robert", "302-555-1234");
         Assert.assertNotNull(book);
-
-
-
     }
+
     @Test
     public void testConstructorWithArgument() {
-
+        PhoneBook book = new PhoneBook();
+        book.add("Bob", "302-555-1234");
+        Assert.assertNotNull(book);
     }
 
     @Test
     public void testAdd() {
+
+        //Given
         PhoneBook book = new PhoneBook();
+        String expectedAddition = "302-555-1234";
+
+        //When
         book.add("Bob", "302-555-1234");
+        String actualAddition = book.lookup("Bob");
 
+        //Then
+        Assert.assertEquals(expectedAddition, actualAddition);
 
     }