Add WK_ARRAY and WK_SET annotations