idk why i haven't done this before

PhoneBook.java

@@ -1,5 +1,6 @@
 import java.util.Scanner;
 import java.util.ArrayList;
+import java.util.*;
 
 /**
  * Created by leon on 1/23/18.
@@ -7,23 +8,45 @@ import java.util.ArrayList;
 public class PhoneBook {
     private String name;
     private String phoneNumber;
-    
+    TreeMap<String, String>phonebook = new TreeMap<>(); 
+
     public PhoneBook()
-    {
-        
+    {        
     }
     
-    public String add(String n, String number)
-    {
-        n = name;
-        number = phoneNumber;
-        
-        return null;
+    public PhoneBook(String name, String phoneNumber) {
+    this.name = name; this.phoneNumber = phoneNumber; 
+    }
+
+    public PhoneBook (TreeMap<String, String>phonebook) {
+        this.phonebook = phonebook;
+    }
+
+    public void add(String name, String phoneNumber) {
+        phonebook.put(name, phoneNumber); 
+    }
+
+    public void remove(String name) {
+        phonebook.remove(name); 
     }
     
-    public String lookUp(String n)
-    {
-        n = name;
-        return null;
+    public void removeRecord() {
+     phonebook.clear();   
     }
+
+    public Collection<String> lookup (String name) {
+        return phonebook.values();   
+    }
+
+    public Collection<String> reverseLookup(String phoneNumber) {
+        return phonebook.keySet();
+    }
+
+    public TreeMap<String, String> display() {
+        return phonebook; 
+    }
+
+    public void entries () {
+
+    } 
 }

PhoneBookTest.java

@@ -2,6 +2,8 @@ import static org.junit.Assert.*;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
+import java.util.*;
+import java.util.stream.Collectors;
 
 /**
  * The test class PhoneBookTest.
@@ -11,45 +13,79 @@ import org.junit.Test;
  */
 public class PhoneBookTest
 {
-    /**
-     * Default constructor for test class PhoneBookTest
-     */
-    public PhoneBookTest()
+    TreeMap<String, String> phonebook = new TreeMap<>(); 
+    PhoneBook pb = new PhoneBook(phonebook);
+
+    @Test
+    public void testAdd()
     {
+        //plug in a String name here using methods from main class
+        String name = "Ryan"; 
+        String phoneNumber= "123-456-7890"; 
+        pb.add(name, phoneNumber);
+        assertFalse(phonebook.isEmpty());
+
     }
-    private PhoneBook phoneBook;
-
-    /**
-     * Sets up the test fixture.
-     *
-     * Called before every test case method.
-     */
-    @Before
-    public void setUp()
-    {
-        phoneBook = new PhoneBook();
+
+    @Test
+    public void testRemove() 
+    {   
+        String name = "Ryan"; 
+        String phoneNumber= "123-456-7890"; 
+
+        pb.add(name, phoneNumber);
+        pb.remove(name);
+        assertTrue(phonebook.isEmpty());
+
+    }
+    
+       @Test
+    public void testRemoveRecord()
+    {   
+        String name = "Ryan"; 
+        String phoneNumber= "123-456-7890"; 
+
+        pb.add(name, phoneNumber);
+        pb.remove(name);
+        assertTrue(phonebook.isEmpty());
+
     }
 
     @Test
-    public void testAdd()
+    public void testLookup()
     {
-        //plug in a String name here using methods from main class
         String name = "Ryan"; 
-        String expectedPhoneNumber= "123-456-7890"; 
-        phoneBook.add(name, expectedPhoneNumber);
-        String actualPhoneNumber = phoneBook.lookUp(name);
-        
-        assertEquals(actualPhoneNumber, expectedPhoneNumber);
-        
+        String phoneNumber= "123-456-7890"; 
+        pb.add(name, phoneNumber);
+        Collection<String> expected = phonebook.values(); 
+
+        assertEquals(expected, pb.lookup(name));
     }
 
-    /**
-     * Tears down the test fixture.
-     *
-     * Called after every test case method.
-     */
-    @After
-    public void tearDown()
+    @Test
+    public void testReverseLookup() {
+        String name = "Ryan"; 
+        String phoneNumber= "123-456-7890"; 
+        pb.add(name, phoneNumber);
+        Collection<String> expected = phonebook.keySet(); 
+
+        assertEquals(expected, pb.reverseLookup(phoneNumber));
+    }
+
+    @Test
+    public void testDisplay() 
+    {       
+        String name = "Ryan"; 
+        String phoneNumber= "123-456-7890"; 
+        pb.add(name, phoneNumber);
+        TreeMap<String, String> expected = phonebook; 
+        
+        assertEquals(expected, pb.display());
+    }
+    
+    @Test
+    public void testEntries()
     {
+        assertEquals(null, 1);  
     }
 }

package.bluej

@@ -2,10 +2,10 @@
 dependency1.from=PhoneBookTest
 dependency1.to=PhoneBook
 dependency1.type=UsesDependency
-editor.fx.0.height=722
-editor.fx.0.width=800
-editor.fx.0.x=240
-editor.fx.0.y=26
+editor.fx.0.height=0
+editor.fx.0.width=0
+editor.fx.0.x=0
+editor.fx.0.y=0
 objectbench.height=101
 objectbench.width=461
 package.divider.horizontal=0.6