Change class to module