Added temp fix, since reloading tree seemed like the only way it worked.  I reload the tree when you hit "Enter" indicating that you are done drawing.