Wrapped the functionality in a class