update for extra credit HW
update for extra credit HW