Add optional key pruning
Update documentation