fix a problem in question 4