HW9: Slight changes