I want to annotate function, tat takes array with at least key "a" and returns the same array