Implement portfolio optimization simulation | DRW Interview Question