clean version with cs225 lab_tree added