Add onEnable method