Add wrapper class