added a class wrapper