DEV: Add function to print tree as level