changed class to module