Update chain.py