Update treeforces.cpp