hw3; q10: fix solution