Allow custom function for drawing tree_icon, revert my last commit