Add key edition feature