Added the portfolio logic