Problem 6
Update problem 6