Complete exercise 1.9

aldreth · aldreth · commit 2bf019caa10b · 2024-03-28T15:18:01.000Z
diff --git a/Work/mortgage.py b/Work/mortgage.py
@@ -8,10 +8,14 @@
 total_paid = 0.0
 months = 0
 
+extra_payment_start_month = int(input('Which month should extra payment start in?'))
+extra_payment_end_month = int(input('Which month should extra payment end in?'))
+extra_payment = int(input('How much extra payment per month can you afford?'))
+
 while principal > 0:
     months = months + 1
-    if months <= 12:
-        principal = principal * (1+rate/12) - payment - 1000
+    if months > extra_payment_start_month and months <= extra_payment_end_month:
+        principal = principal * (1+rate/12) - payment - extra_payment
     else:
         principal = principal * (1+rate/12) - payment
       
