Update investment_calculator.py