update homework5 of cs229