Added on_init hook
on_init hook