Added init_module and cleanup_module