Add annotations for function types