Add prop type validation to Key component