Added initial_investment variable