Add key map for saving