Add the unlock fonctionality