Now adding a wrapper class which adds more flexibility, still need to update README.md