Update encoder.py