tree drawing cleanup but not completely working.