Update transformer.py