Update with original code from HW6 in hope to get rid of errors