Clean up Best time to buy and sell stock II