Add unregister function