diff --git a/pom.xml b/pom.xml
index b8ae243..2506475 100644
--- a/pom.xml
+++ b/pom.xml
@@ -6,7 +6,7 @@
 
     <groupId>nl.andrewlalis</groupId>
     <artifactId>EntityRelationMappingEditor</artifactId>
-    <version>1.3.1</version>
+    <version>1.4.0</version>
     <build>
         <plugins>
             <plugin>
diff --git a/src/main/java/nl/andrewlalis/erme/EntityRelationMappingEditor.java b/src/main/java/nl/andrewlalis/erme/EntityRelationMappingEditor.java
index b9cafa6..9341024 100644
--- a/src/main/java/nl/andrewlalis/erme/EntityRelationMappingEditor.java
+++ b/src/main/java/nl/andrewlalis/erme/EntityRelationMappingEditor.java
@@ -7,7 +7,7 @@ import nl.andrewlalis.erme.view.EditorFrame;
 import java.nio.charset.StandardCharsets;
 
 public class EntityRelationMappingEditor {
-	public static final String VERSION = "1.3.1";
+	public static final String VERSION = "1.4.0";
 
 	public static void main(String[] args) {
 		if (!FlatLightLaf.install()) {
diff --git a/src/main/java/nl/andrewlalis/erme/control/actions/LolcatAction.java b/src/main/java/nl/andrewlalis/erme/control/actions/LolcatAction.java
index f2ffc74..5b10809 100644
--- a/src/main/java/nl/andrewlalis/erme/control/actions/LolcatAction.java
+++ b/src/main/java/nl/andrewlalis/erme/control/actions/LolcatAction.java
@@ -2,9 +2,7 @@ package nl.andrewlalis.erme.control.actions;
 
 import lombok.Getter;
 import lombok.Setter;
-import nl.andrewlalis.erme.EntityRelationMappingEditor;
 import nl.andrewlalis.erme.view.DiagramPanel;
-import nl.andrewlalis.erme.view.view_models.AttributeViewModel;
 
 import javax.swing.*;
 import java.awt.event.ActionEvent;
diff --git a/src/main/java/nl/andrewlalis/erme/view/view_models/AttributeViewModel.java b/src/main/java/nl/andrewlalis/erme/view/view_models/AttributeViewModel.java
index 48617ed..754bd3e 100644
--- a/src/main/java/nl/andrewlalis/erme/view/view_models/AttributeViewModel.java
+++ b/src/main/java/nl/andrewlalis/erme/view/view_models/AttributeViewModel.java
@@ -1,11 +1,9 @@
 package nl.andrewlalis.erme.view.view_models;
 
-import nl.andrewlalis.erme.EntityRelationMappingEditor;
 import nl.andrewlalis.erme.control.actions.LolcatAction;
 import nl.andrewlalis.erme.model.Attribute;
 import nl.andrewlalis.erme.model.AttributeType;
 import nl.andrewlalis.erme.model.ForeignKeyAttribute;
-import nl.andrewlalis.erme.view.DiagramPanel;
 
 import java.awt.*;
 import java.awt.font.TextAttribute;
@@ -52,8 +50,6 @@ public class AttributeViewModel implements ViewModel {
 		Point offset = g.getClipBounds().getLocation();
 		g.translate(offset.x, offset.y);
 		Dimension viewportSize = g.getClipBounds().getSize();
-		g.drawRect(0, 0, viewportSize.width, viewportSize.height);
-		g.fillRect(-5, -5, 10, 10);
 
 		double diagonal_slope = (double) viewportSize.width / (double) viewportSize.height;
 		double perp_slope = -1f / diagonal_slope;