add wrapper class