Adding a wrapper class