Removing recursion from find_mut in treemap