added changes for HW4 pt1