completed exercise 2, began bonus, need to balance trees