add return types to ast function outputs