while true do
    if turtle.compare(2) then
        turtle.select(1)
        turtle.place()

        turtle.dig()
        turtle.forward();
        -- Dig up and count tree height to get back to ground
        local height = 0
        while turtle.inspectUp() do
            turtle.digUp()
            turtle.up()
            height = height + 1
        end

        -- Get back to ground
        for i = 0, height, 1 do
            turtle.down()
        end

        -- Move back
        turtle.back()
    end
end