Removed unnecessary method and moved the module to a class.