HW 7: added part B