Add wrapper?