Adding the wrapper class to the readme