Added method to remove keys and also write operations now return the status