Updated comp sci homework