Add an initialization function