modifications for homework6