1 files changed, 4 insertions, 5 deletions
diff --git a/court.mjs b/court.mjs
index aa5bf29..713c488 100644
--- a/court.mjs
+++ b/court.mjs
@@ -15,9 +15,8 @@ class Court {
     console.log(this.name + " is currently " + status);
   }
   render() {
-    let type = this.status
-      ? `class="button is-medium is-success"`
-      : 'class="button is-medium is-danger"';
+    let color = this.status ? "is-success" : "is-danger";
+    let type = `class="button is-medium ${color}"`;
     return `<button id=${"court_" + this.id} ${type}>${this.name}</button>`;
   }
 }
@@ -39,7 +38,7 @@ export class CourtManager extends Manager {
     }
     console.log("Loaded " + this.listData.length + " courts.");
     this.render();
-    this.addToggleBehaviour();
+    this.addBehaviour();
   }
   generate() {
     console.log("Generating court data.");
@@ -49,6 +48,6 @@ export class CourtManager extends Manager {
     console.log("Generated data for " + this.listData.length + " courts");
     this.store();
     this.render();
-    this.addToggleBehaviour();
+    this.addBehaviour();
   }
 }