Added initialization function