setelah click maka tidak bisa kembali lagi

saya ingin jika tombol yang memiliki class (.lock) di click maka class di dalamnya akan di tambahkan class baru (.unLock) dan class yang sebelumnya akan di hapus, tetapi setelah berubah saya ingin kembali class sebelumnya (.lock) malah gk bisa. tolong bantuannya

  $('.lock').click(function(){
    var id = $(this).attr('id');
    $('.lock'+id).addClass("fa-unlock btn btn-success unLock unLock"+id);
    $('.lock'+id).removeClass("fa-lock btn-danger lock lock"+id);
  });
  $('.unLock').click(function(){
    var id = $(this).attr('id');
    $('.unLock'+id).addClass("fa-lock btn btn-danger lock lock"+id);
    $('.unLock'+id).removeClass("fa-unlock btn-success unLock unLock"+id);
  });