Add after_initialize hook