Add onEnable() and onDisable()