Add features to transform.py