moved code to draw "choose tree" screen to separate function