Adding the correct key on the code