Wrote code and modified comments