Add code for buy/sell stock.