the real final commit

PhoneBook.java

@@ -17,6 +17,7 @@ public class PhoneBook {
     public void remove (String phoneNumber) {
         ArrayList<String> edit = new ArrayList<String>();
         Set<String> keys = phoneBook.keySet();
+        
         for (String key : keys) {
             if (phoneBook.get(key).contains(phoneNumber)) {
                 edit = phoneBook.get(key);
@@ -31,12 +32,11 @@ public class PhoneBook {
     }
     
     public String lookup(String name) {
-        
-        return phoneBook.containsKey(name) ?
-               phoneBook.get(name)
-                        .toString()
-                        .replaceAll("[\\[\\]]", "")
-            : "N/A";
+        return phoneBook.containsKey(name) ? format(phoneBook.get(name)): "N/A";
+    }
+    
+    public String format(ArrayList<String> numbers) {
+        return numbers.toString().replaceAll("[\\[\\]]", "");
     }
     
     public String reverseLookup(String phoneNumber) {

PhoneBookTest.java

@@ -1,5 +1,4 @@
 
-
 import static org.junit.Assert.*;
 import org.junit.After;
 import org.junit.Before;
@@ -46,6 +45,19 @@ public class PhoneBookTest
     }
     
     @Test
+    public void testReverseLookup()
+    {
+       PhoneBook test = new PhoneBook();
+       String phoneNumber = "(302)-867-5309";
+       String expectedName = "Jenny";
+       test.add(expectedName, phoneNumber);
+       
+       String actual = test.reverseLookup(phoneNumber);
+       assertEquals(expectedName, actual);
+     
+    } 
+    
+    @Test
     public void testRemoveIndividualNumber() {
        PhoneBook test = new PhoneBook();
        String name = "Jenny";
@@ -54,26 +66,15 @@ public class PhoneBookTest
        test.add(name, number);
        test.add(name, number2);
        test.remove(number);
+       
        String actual = test.reverseLookup(number);
        assertEquals("N/A", actual);
     }
     
     @Test
-    public void testReverseLookup()
+    public void testDisplay() 
     {
        PhoneBook test = new PhoneBook();
-       String phoneNumber = "(302)-867-5309";
-       String expectedName = "Jenny";
-       test.add(expectedName, phoneNumber);
-       
-       String actual = test.reverseLookup(phoneNumber);
-       assertEquals(expectedName, actual);
-     
-    } 
-    
-    @Test
-    public void testDisplay() {
-       PhoneBook test = new PhoneBook();
        test.add("hey", "302");
        test.add("yo", "521");
        test.add("hey", "498");