AST: Type annotations

Signed-off-by: Pekka Enberg <penberg@iki.fi>