Tree Python Code