HW6 for CodeFellows