Add key length check