Add models/investment.py