add homeork 5 problem 3