Update to outdated code for portfolio