change class param to module, class might be used later