Changes for HW4.