better tree drawing (with some hacks)