[ASDL cleanup] Add some type annotations