updated hw05 revised the problem