Update models.py

add investment types