HW5 code complete