HW5 for CodeFellows