Adds valid key checking