Cleaning up style for Tree class