started modifications for HW 4