Add wrapper function