added trees in the training screen