Add getKey method