fix labeling, add comments