Adding ability to visualize tree extensions. Adding back in obstacles.