add parsing for argmin calls, add ast rewrite to update stratification info