diff --git a/src/main/java/no/ntnu/idatt1002/demo/controller/BudgetController.java b/src/main/java/no/ntnu/idatt1002/demo/controller/BudgetController.java
index edcb0d1d1ac3071b2aa02d8fc95ff30e6434c68d..d8f1335652f7d2d7b5a0286cf1f6bbb7d8594566 100644
--- a/src/main/java/no/ntnu/idatt1002/demo/controller/BudgetController.java
+++ b/src/main/java/no/ntnu/idatt1002/demo/controller/BudgetController.java
@@ -25,8 +25,7 @@ import java.util.Optional;
 public class BudgetController {
 
     private DialogMode dialogMode;
-    private List<BudgetItem> listOfBudgetItems = new ArrayList<>();
-    private GeneralBudget general = new GeneralBudget(31, listOfBudgetItems, 200);
+    private GeneralBudget general = new GeneralBudget(31, 2000);
 
 
     @FXML
@@ -65,7 +64,7 @@ public class BudgetController {
 
 
     public void initialize() throws IOException {
-        budgetList = FXCollections.observableArrayList(listOfBudgetItems);
+        budgetList = FXCollections.observableArrayList(general.getBudgetItems());
         budgetTableView.setItems(budgetList);
 
         categoryColumn.setCellValueFactory(new PropertyValueFactory<BudgetItem, ExpenseCategory>("budgetCategory"));
@@ -108,7 +107,7 @@ public class BudgetController {
         item = budgetController.getNewBudgetItem();
         if(item != null && dialogMode == DialogMode.ADD){
             try {
-            general.addToBudget(item.getBudgetAmount(), item.getBudgetDescription(), item.getBudgetCategory());
+            general.addToBudgetBudgetItem(item);
             } catch(IllegalArgumentException e) {
                 showIllegalBudgetItemDialog();
             }
@@ -149,7 +148,7 @@ public class BudgetController {
         Alert alert = new Alert(AlertType.ERROR);
         alert.setTitle("Budget amount exceeded");
         alert.setHeaderText("Your budget exceeds the max limit");
-        alert.setContentText("The total budget sum must be bl" );
+        alert.setContentText("The total budget sum must be below " + general.getMaxAmount());
 
         return alert.showAndWait();
     }
@@ -157,7 +156,7 @@ public class BudgetController {
 
     protected void refreshObservableList(){
         budgetTableView.setItems(budgetList);
-        this.budgetList.setAll(listOfBudgetItems);
+        this.budgetList.setAll(general.getBudgetItems());
     }
 
 }