HW6 without extra
Final hw6