diff --git a/budget.py b/budget.py
index 8f3b677..066ea2c 100644
--- a/budget.py
+++ b/budget.py
@@ -464,7 +464,7 @@ class BalanceShow:
 
                 row = cur.fetchone()
 
-                incomes = row[0]
+                incomes = row[0] if row[0] != None else 0
 
                 cur.execute(bills_sql, (self.criteria,))
 
@@ -472,6 +472,8 @@ class BalanceShow:
 
                 bills = row[0]
 
+                bills = row[0] if row[0] != None else 0
+
             else:
 
                 cur.execute(incomes_sql)
@@ -676,6 +678,8 @@ if __name__ == '__main__':
 
                 print_table.add_row([b_amount])
 
+                #print_table.add_row(['Total', round(total_amount, 2)])
+
                 print(print_table)
 
         elif option == 6: