Homework 6 Fix