Raw press of edit and expense buttons will not refresh pie charts

src/main/java/no/ntnu/idatt1002/demo/controller/IncomeExpenseController.java

@@ -209,14 +209,21 @@ public class IncomeExpenseController implements FinanceController {
    */
   @Override
   public void handleAddBtn(javafx.event.ActionEvent event) {
+    int sizeBf = (expenseRegister.getItems().size() + incomeRegister.getItems().size());
+
     if (event.getSource() == addIncome) {
       handleAddIncome();
     } else if (event.getSource() == addExpense){
       handleAddExpense();
     }
-    refreshTableView();
-    refreshPieCharts();
-    refreshProgress();
+
+    int sizeAf = (expenseRegister.getItems().size() + incomeRegister.getItems().size());
+
+    if (sizeAf != sizeBf) {
+      refreshTableView();
+      refreshPieCharts();
+      refreshProgress();
+    }
   }
 
   /**
@@ -234,14 +241,14 @@ public class IncomeExpenseController implements FinanceController {
         handleEditIncome(chosenIncome);
       } else if (chosenExpense != null) {
         handleEditExpense(chosenExpense);
-      }
+      } else return;
     } else if (event.getSource() == deleteBtn) {
       if (chosenIncome != null) {
         handleDeleteIncome(chosenIncome);
       } else if (chosenExpense != null) {
         handleDeleteExpense(chosenExpense);
-      }
-    } else return;
+      } else return;
+    }
     //Updates the tableview and pie chart using the register
     refreshTableView();
     refreshPieCharts();