extract identifyCompany method

1 files changed, 17 insertions, 16 deletions

diff --git a/src/Q3/CreditCard.java b/src/Q3/CreditCard.java
index e82eb6c..1146129 100644
--- a/src/Q3/CreditCard.java
+++ b/src/Q3/CreditCard.java
@@ -11,22 +11,7 @@ public class CreditCard {
 
   public CreditCard(String num) {
     this.ccNumber = num;
-    if (validateCompany()) {
-      switch (num.charAt(0)) {
-        case '3':
-          this.company = "American Express";
-          break;
-        case '4':
-          this.company = "Visa";
-          break;
-        case '5':
-          this.company = "MasterCard";
-          break;
-        case '6':
-          this.company = "Discover";
-          break;
-      }
-    }
+    this.company = this.identifyCompany(num);
   }
 
   public int getEvenSum() {
@@ -75,4 +60,20 @@ public class CreditCard {
   public boolean validateSums() {
     return false;
   }
+
+  private String identifyCompany(String number) {
+    if (validateCompany())
+      switch (number.charAt(0)) {
+        case '3':
+          return "American Express";
+        case '4':
+          return "Visa";
+        case '5':
+          return "MasterCard";
+        case '6':
+          return "Discover";
+      }
+
+    return "Unknown";
+  }
 }