Add after_initialize for models