I am not sure that's a good idea. Now phones are rarely stolen. I would not like to come back to time when you get mugged for your phone. (1st hand experience)
In fact that is how most phones (that still have an unlockable bootloader) do it, the button is in the developer options which if you reset the phone you cannot get to without having access to the linked Google account.