Update encode.py