Add module initialization