Best Time to Buy and Sell Stock II: cleaner code