adjust files according to HW3 description