Add location information to ast