Working! Added some stress test and missing cases when removing nodes.