Updated wrapper function