Update tf.py