Add ability to partially replace keys