Update compare_stock.py