added a class inside the module instead