Add ability to update key