diff --git a/src/Regex.java b/src/Regex.java
index a5e38c6..5fa96a1 100644
--- a/src/Regex.java
+++ b/src/Regex.java
@@ -50,5 +50,5 @@ public class Regex {
      *     <li>2 The datatype</li>
      *  </ul>
      */
-    public static final String ARG_SPLIT_REGEX = " ?(\\w+): (\\w+)";
+    public static final String ARG_SPLIT_REGEX = " *(\\w+) *: *(\\w+)";
 }
diff --git a/src/UMLClass.java b/src/UMLClass.java
index 3298da4..ad101aa 100644
--- a/src/UMLClass.java
+++ b/src/UMLClass.java
@@ -52,8 +52,6 @@ public class UMLClass {
             for (UMLField field : fields) {
                 s.append(field.toString());
             }
-
-            s.append("\n");
         }
 
         if (methods.size() > 0) {
diff --git a/src/UMLMethod.java b/src/UMLMethod.java
index 031b331..53df229 100644
--- a/src/UMLMethod.java
+++ b/src/UMLMethod.java
@@ -50,7 +50,7 @@ public class UMLMethod {
     @Override
     public String toString() {
         StringBuilder returnString = new StringBuilder();
-        returnString.append("   ").append(encapsulation).append(returnType).append(name).append(" (");
+        returnString.append("\n   ").append(encapsulation).append(returnType).append(name).append(" (");
 
         for (int i = 0; i < argsNames.size(); i++) {
             returnString.append(argsTypes.get(i)).append(" ").append(argsNames.get(i));