diff --git a/main.py b/main.py
index 66790f0..3ab33a5 100644
--- a/main.py
+++ b/main.py
@@ -185,14 +185,23 @@ def start():
     startProcs()
 
 def memUsage():
-    a = subprocess.check_output("free -m".split()).decode().split("\n")[1]
+    aa = subprocess.check_output("free -m".split()).decode().split("\n")
+
+    a = aa[1]
 
     while "  " in a:
         a = a.replace("  ", " ")
     
     ae = a.split()
 
-    return [int(e) for e in [ae[2], ae[4], ae[5], ae[6]]]
+    af = aa[2]
+
+    while "  " in a:
+        af = af.replace("  ", " ")
+    
+    af = af.split()
+
+    return [int(e) for e in [ae[2], ae[4], ae[5], ae[6], af[2], af[3]]]
 
 def cpuUsage():
     a = subprocess.check_output(["bash", "-c", "top -b -n1 | grep \"Cpu(s)\" | awk '{print $2+$4}'"]).decode().rstrip("\n")
diff --git a/templates/main.html b/templates/main.html
index a0264eb..026f8f6 100644
--- a/templates/main.html
+++ b/templates/main.html
@@ -8,14 +8,18 @@
     </div>
 
     <script>   
+        Chart.overrides['doughnut'].plugins.legend.display = false;
+
         const mconfig = {
             type: 'doughnut',
             data: {
                 labels: [
-                    "used",
-                    "shared",
-                    "buff/cache",
-                    "available"
+                    "Used",
+                    "Shared",
+                    "Buff/Cache",
+                    "Available",
+                    "Swap Used",
+                    "Swap Free"
                 ],
                 datasets: [
                     {
@@ -23,7 +27,9 @@
                             '#005C46',
                             "#009B77",
                             "#34AC90",
-                            '#00000000'
+                            '#00000000',
+                            '#7339ac',
+                            '#7339ac15'
                         ],
                         data: [],
                     }