Transformed the module to a class and added new tests