diff --git a/.idea/misc.xml b/.idea/misc.xml
index 2a824a2..e69afb6 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
-  <component name="ProjectRootManager" version="2" languageLevel="JDK_14" default="true" project-jdk-name="14" project-jdk-type="JavaSDK">
+  <component name="ProjectRootManager" version="2" languageLevel="JDK_14" default="false" project-jdk-name="openjdk-14" project-jdk-type="JavaSDK">
     <output url="file://$PROJECT_DIR$/out" />
   </component>
 </project>
\ No newline at end of file
diff --git a/src/UMLMethod.java b/src/UMLMethod.java
index 5b28f33..95b80d2 100644
--- a/src/UMLMethod.java
+++ b/src/UMLMethod.java
@@ -34,9 +34,9 @@ public class UMLMethod {
 
 
         if(parts[1].equals("") && !className.equals(name)){
-            this.returnType = "void";
+            this.returnType = "void ";
         } else {
-            this.returnType = parts[1];
+            this.returnType = parts[1] + " ";
         }
 
         if(parts.length == 4) {
@@ -57,7 +57,7 @@ public class UMLMethod {
     @Override
     public String toString() {
         StringBuilder returnString = new StringBuilder();
-        returnString.append("\n   ").append(encapsulation).append(returnType).append(" ").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));