Edited my personal code to use new tree naming