Better consistency for key? method

This makes sense, feel free to open a PR :slightly_smiling_face: