Add locations to AST elements to prepare for API annotations