Submission-Ready CS169 Warmup 1 Code