Got stock.py working