Adding Functionality to Set the Key