added type to investment model