fixing my change from HW4