Update stocky.py